import os
import torch
from configure import configure
from data_utils import Dataer
from network import Network
import numpy as np
import time
from utils import Print_config

if __name__ == "__main__":

    config = vars(configure())

    os.environ['CUDA_VISBLE_DEVICES'] = '{}'.format(config['gpu_id'])
    device = 'cuda' if torch.cuda.is_available() else 'cpu'
    config['device'] = device

    Print_config(config=config)


    dataer = Dataer(dataset_name=config['dataset'])

    neter = Network(dataer, config)
    neter.Test_model()